# -*- coding: utf-8 -*-
"""
------------------------------------------------------------------------------
    File Name:  svd_demo
    Author   :  wanwei1029
    Date     :  2018/11/20
    Desc     :
------------------------------------------------------------------------------
"""
import numpy as np


def svd_demo():
    data = [[0,0,0,2,2],
                [0,0,0,3,3],
                [0,0,0,1,1],
                [1,1,1,0,0],
                [2,2,2,0,0],
                [5,5,5,0,0],
                [1,1,1,0,0]]

    u, sigma, vt = np.linalg.svd(data)
    print(sigma)


def demo():
    svd_demo()


if __name__ == '__main__':
    test_method = "demo"
    if test_method == "demo":
        demo()
